import numpy as np

n1 = np.array([[1, 2], [3, 4], [5, 6]])
# 创建第二个数组
n2 = np.array([[10, 20], [30, 40], [50, 60]])
print(n1)
print(n2)

"""
1、数组的增加
"""
# 水平方向增加数据 -- 列的个数增加
print(np.hstack((n1, n2)))

# 垂直方向增加数据 -- 行的个数增加
print(np.vstack((n1, n2)))

"""
2、数组的删除
"""
n1 = np.array([[1, 2], [3, 4], [5, 6]])
print(n1)

# 删除第3行 -- 0号轴，索引为2的数据
n2 = np.delete(n1, 2, axis=0)
print(n2)

# 删除第1列
n3 = np.delete(n1, 0, axis=1)
print(n3)

# 删除第1行和第3行
n4 = np.delete(n1, (0, 2), axis=0)
print(n4)

"""
3、数组的修改
"""
n1 = np.array([[1, 2], [3, 4], [5, 6]])
n1[1] = [30, 40]  # 修改一行
print(n1)

n1[2][1] = 88  # 修改单个元素
print(n1)

"""
4、数组的查询
"""
n = np.arange(1, 11)
print(n)

n2 = n[np.where(n > 5)]  # 获取到数组中所有大于5的元素, 返回索引
print(n2)

n3 = np.where(n > 5, 2, 0) # 数组中元素大于5，输出2，否则输出0
print(n3)
